Welcome to Hands on deep learning!

02.02.03 转换为张量格式

现在inputs和outputs中的所有条目都是数值类型,它们可以转换为张量格式。 当数据采用张量格式后,可以通过在 2.1节中引入的那些张量函数来进一步操作。

import torch,pandas as pd

data = pd.read_csv(r"D:\Pyobject2023\object\DeepLearning\data\house_tiny.csv")

print(data)

inputs, outputs = data.iloc[:, 0:2], data.iloc[:, 2]

inputs = inputs.fillna(inputs.mean(numeric_only=True)) #numeric_only=True不加会报错,这行不加也会报错

inputs = pd.get_dummies(inputs, dummy_na=True) #这行不加也会报错

X = torch.tensor(inputs.to_numpy(dtype=float))

y = torch.tensor(outputs.to_numpy(dtype=float))

print(X,y) #要注意大小写,不然会报错

返回值:

NumRooms Alley Price

0 NaN Pave 127500

1 2.0 NaN 106000

2 4.0 NaN 178100

3 NaN NaN 140000

tensor([[3., 1., 0.],

[2., 0., 1.],

[4., 0., 1.],

[3., 0., 1.]], dtype=torch.float64) tensor([127500., 106000., 178100., 140000.], dtype=torch.float64)